from shell.aishell import AiShell
from core.agent_factory import create_agent
from core.base_agent import Agent

def main():
    shell = AiShell()
    
    # 使用工厂创建Agent
    agent = create_agent("ai")
    code_agent = create_agent("code")
    
    if agent is None:
        print("错误: 无法创建普通Agent")
        return
    
    if code_agent is None:
        print("错误: 无法创建代码Agent")
        return
    
    shell.setAgent(agent)
    shell.setCodeAgent(code_agent)
    
    print("欢迎使用AITA助手系统！")
    print("可用模式:")
    print("  普通模式 - 默认AI助手")
    print("  代码模式 - 输入 'code' 切换到代码编写助手")
    print("  输入 'help' 查看更多命令")
    
    shell.cmdloop()

if __name__ == "__main__":
    main()